package com.xq.utils;

import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder;
import com.aliyun.oss.model.*;
import com.xq.entity.AliOss;
import org.joda.time.DateTime;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

import java.io.*;
import java.util.Date;
import java.util.UUID;

/**
 * @title: Ossutil
 * @Author Tan
 * @Date: 2020/8/1 14:54
 * @Version 1.0
 */
@Component
public class OssUtil {
    @Autowired
    private AliOss aliOss;

    /**
     * 文件上传
     * @param multipartFile
     * @throws IOException
     */
    public String uploadFile(MultipartFile multipartFile) throws IOException {
        //返回的路径
        String fileUrl=null;
        //获取阿里云存储相关常量
        String endPoint = aliOss.getEndpoint();
        String accessKeyId = aliOss.getAccessKeyId();
        String accessKeySecret = aliOss.getAccessKeySecret();
        String bucketName = aliOss.getBucketName();
        // 创建OSSClient实例。
        OSS ossClient = new OSSClientBuilder().build(endPoint, accessKeyId, accessKeySecret);
        //判断是否存在Bucket实例 不存在就创建
        if (!ossClient.doesBucketExist(bucketName)) {
            //创建bucket
            ossClient.createBucket(bucketName);
            //设置oss实例的访问权限：公共读
            ossClient.setBucketAcl(bucketName, CannedAccessControlList.PublicRead);
        }

        //得到文件流
        InputStream inputStream=multipartFile.getInputStream();
        //格式日期 作为路径
        String datePath = new DateTime().toString("yyyy/MM/dd");

        //原来的文件名
        String original = multipartFile.getOriginalFilename();
        //随机文件名
        String fileName = UUID.randomUUID().toString().replaceAll("-","");
        //截取原来文件名的后缀
        String fileType = original.substring(original.lastIndexOf("."));
        //拼接新的文件名称
        String newName = fileName + fileType;

        //用于拼接oss实例的存储路径
        fileName = datePath+"/"+newName;

        ObjectMetadata objectMetadata = new ObjectMetadata();
        //图片范围 公共读
        objectMetadata.setObjectAcl(CannedAccessControlList.PublicRead);
        //截取图片名称后缀成为 图片内容的类型
        objectMetadata.setContentType(getcontentType(fileName.substring(fileName.lastIndexOf("."))));

        //文件上传至阿里云
        ossClient.putObject(bucketName, fileName, inputStream,objectMetadata);

        // 关闭OSSClient。
        ossClient.shutdown();

        Date expiration = new Date(System.currentTimeMillis() + 3600L * 1000 * 24 * 365 * 10);
        fileUrl = ossClient.generatePresignedUrl(bucketName, fileName, expiration).toString();
        //返回路经地址 其中防止了暴露key
        return  fileUrl.substring(0, fileUrl.indexOf("?"));
    }

    /**
     * 流式下载
     */
    public void downloadIOFile() throws IOException {
        // 创建OSSClient实例。
        OSS ossClient = new OSSClientBuilder().build(aliOss.getEndpoint(), aliOss.getAccessKeyId(), aliOss.getAccessKeySecret());

        // ossObject包含文件所在的存储空间名称、文件名称、文件元信息以及一个输入流。
        OSSObject ossObject = ossClient.getObject(aliOss.getBucketName(), "a.jpg");

        // 读取文件内容。
        System.out.println("Object content:");
        BufferedReader reader = new BufferedReader(new InputStreamReader(ossObject.getObjectContent()));
        while (true) {
            String line = reader.readLine();
            if (line == null) {
                break;
            }

            System.out.println("\n" + line);
        }
        // 数据读取完成后，获取的流必须关闭，否则会造成连接泄漏，导致请求无连接可用，程序无法正常工作。
        reader.close();
        // 关闭OSSClient。
        ossClient.shutdown();
    }

    /**
     * 下载到本地
     */
    public void downloadLocalFile(){
        // 创建OSSClient实例。
        OSS ossClient = new OSSClientBuilder().build(aliOss.getEndpoint(), aliOss.getAccessKeyId(), aliOss.getAccessKeySecret());

        // 下载OSS文件到本地文件。如果指定的本地文件存在会覆盖，不存在则新建。
        ossClient.getObject(new GetObjectRequest(aliOss.getBucketName(), "a.jpg"), new File("G://a.jpg"));

        // 关闭OSSClient。
        ossClient.shutdown();
    }

    /**
     * 文件处理
     * @param FilenameExtension
     * @return
     */
    public static String getcontentType(String FilenameExtension) {
        if (FilenameExtension.equalsIgnoreCase(".bmp")) {
            return "image/bmp";
        }
        if (FilenameExtension.equalsIgnoreCase(".gif")) {
            return "image/gif";
        }
        if (FilenameExtension.equalsIgnoreCase(".jpeg") ||
                FilenameExtension.equalsIgnoreCase(".jpg") ||
                FilenameExtension.equalsIgnoreCase(".png")) {
            return "image/jpg";
        }
        if (FilenameExtension.equalsIgnoreCase(".html")) {
            return "text/html";
        }
        if (FilenameExtension.equalsIgnoreCase(".txt")) {
            return "text/plain";
        }
        if (FilenameExtension.equalsIgnoreCase(".vsd")) {
            return "application/vnd.visio";
        }
        if (FilenameExtension.equalsIgnoreCase(".pptx") ||
                FilenameExtension.equalsIgnoreCase(".ppt")) {
            return "application/vnd.ms-powerpoint";
        }
        if (FilenameExtension.equalsIgnoreCase(".docx") ||
                FilenameExtension.equalsIgnoreCase(".doc")) {
            return "application/msword";
        }
        if (FilenameExtension.equalsIgnoreCase(".xml")) {
            return "text/xml";
        }
        return "image/jpg";
    }
}